// Service Worker for 养基宝 Chrome Extension - Manifest V3
// 替换原有的background script以兼容Chrome 139+

// Storage API wrapper
const storageAPI = {
    set: function(key, value) {
        return new Promise((resolve) => {
            chrome.storage.local.set({ [key]: value }, () => resolve("success"));
        });
    },
    get: function(key) {
        return new Promise((resolve) => {
            chrome.storage.local.get(key, (result) => resolve(result[key]));
        });
    },
    remove: function(key) {
        return new Promise((resolve) => {
            chrome.storage.local.remove(key, () => resolve("success"));
        });
    }
};

// API配置
const API_SECRET = "YxmKSrQR4uoJ5lOoWIhcbd7SlUEh9OOc";
const API_BASE_URL = "http://browser-plug-api.yangjibao.com";

// 完整的MD5实现
function md5(str) {
    function utf8Encode(str) {
        return unescape(encodeURIComponent(str));
    }
    
    function binl2hex(binarray) {
        const hex_tab = "0123456789abcdef";
        let str = "";
        for (let i = 0; i < binarray.length * 4; i++) {
            str += hex_tab.charAt((binarray[i >> 2] >> ((i % 4) * 8 + 4)) & 0xF) +
                   hex_tab.charAt((binarray[i >> 2] >> ((i % 4) * 8)) & 0xF);
        }
        return str;
    }
    
    function str2binl(str) {
        const bin = [];
        const mask = (1 << 8) - 1;
        for (let i = 0; i < str.length * 8; i += 8) {
            bin[i >> 5] |= (str.charCodeAt(i / 8) & mask) << (i % 32);
        }
        return bin;
    }
    
    function binlMD5(x, len) {
        x[len >> 5] |= 0x80 << ((len) % 32);
        x[(((len + 64) >>> 9) << 4) + 14] = len;
        
        let a = 1732584193;
        let b = -271733879;
        let c = -1732584194;
        let d = 271733878;
        
        for (let i = 0; i < x.length; i += 16) {
            const olda = a, oldb = b, oldc = c, oldd = d;
            
            a = md5_ff(a, b, c, d, x[i + 0], 7, -680876936);
            d = md5_ff(d, a, b, c, x[i + 1], 12, -389564586);
            c = md5_ff(c, d, a, b, x[i + 2], 17, 606105819);
            b = md5_ff(b, c, d, a, x[i + 3], 22, -1044525330);
            a = md5_ff(a, b, c, d, x[i + 4], 7, -176418897);
            d = md5_ff(d, a, b, c, x[i + 5], 12, 1200080426);
            c = md5_ff(c, d, a, b, x[i + 6], 17, -1473231341);
            b = md5_ff(b, c, d, a, x[i + 7], 22, -45705983);
            a = md5_ff(a, b, c, d, x[i + 8], 7, 1770035416);
            d = md5_ff(d, a, b, c, x[i + 9], 12, -1958414417);
            c = md5_ff(c, d, a, b, x[i + 10], 17, -42063);
            b = md5_ff(b, c, d, a, x[i + 11], 22, -1990404162);
            a = md5_ff(a, b, c, d, x[i + 12], 7, 1804603682);
            d = md5_ff(d, a, b, c, x[i + 13], 12, -40341101);
            c = md5_ff(c, d, a, b, x[i + 14], 17, -1502002290);
            b = md5_ff(b, c, d, a, x[i + 15], 22, 1236535329);
            
            a = md5_gg(a, b, c, d, x[i + 1], 5, -165796510);
            d = md5_gg(d, a, b, c, x[i + 6], 9, -1069501632);
            c = md5_gg(c, d, a, b, x[i + 11], 14, 643717713);
            b = md5_gg(b, c, d, a, x[i + 0], 20, -373897302);
            a = md5_gg(a, b, c, d, x[i + 5], 5, -701558691);
            d = md5_gg(d, a, b, c, x[i + 10], 9, 38016083);
            c = md5_gg(c, d, a, b, x[i + 15], 14, -660478335);
            b = md5_gg(b, c, d, a, x[i + 4], 20, -405537848);
            a = md5_gg(a, b, c, d, x[i + 9], 5, 568446438);
            d = md5_gg(d, a, b, c, x[i + 14], 9, -1019803690);
            c = md5_gg(c, d, a, b, x[i + 3], 14, -187363961);
            b = md5_gg(b, c, d, a, x[i + 8], 20, 1163531501);
            a = md5_gg(a, b, c, d, x[i + 13], 5, -1444681467);
            d = md5_gg(d, a, b, c, x[i + 2], 9, -51403784);
            c = md5_gg(c, d, a, b, x[i + 7], 14, 1735328473);
            b = md5_gg(b, c, d, a, x[i + 12], 20, -1926607734);
            
            a = md5_hh(a, b, c, d, x[i + 5], 4, -378558);
            d = md5_hh(d, a, b, c, x[i + 8], 11, -2022574463);
            c = md5_hh(c, d, a, b, x[i + 11], 16, 1839030562);
            b = md5_hh(b, c, d, a, x[i + 14], 23, -35309556);
            a = md5_hh(a, b, c, d, x[i + 1], 4, -1530992060);
            d = md5_hh(d, a, b, c, x[i + 4], 11, 1272893353);
            c = md5_hh(c, d, a, b, x[i + 7], 16, -155497632);
            b = md5_hh(b, c, d, a, x[i + 10], 23, -1094730640);
            a = md5_hh(a, b, c, d, x[i + 13], 4, 681279174);
            d = md5_hh(d, a, b, c, x[i + 0], 11, -358537222);
            c = md5_hh(c, d, a, b, x[i + 3], 16, -722521979);
            b = md5_hh(b, c, d, a, x[i + 6], 23, 76029189);
            a = md5_hh(a, b, c, d, x[i + 9], 4, -640364487);
            d = md5_hh(d, a, b, c, x[i + 12], 11, -421815835);
            c = md5_hh(c, d, a, b, x[i + 15], 16, 530742520);
            b = md5_hh(b, c, d, a, x[i + 2], 23, -995338651);
            
            a = md5_ii(a, b, c, d, x[i + 0], 6, -198630844);
            d = md5_ii(d, a, b, c, x[i + 7], 10, 1126891415);
            c = md5_ii(c, d, a, b, x[i + 14], 15, -1416354905);
            b = md5_ii(b, c, d, a, x[i + 5], 21, -57434055);
            a = md5_ii(a, b, c, d, x[i + 12], 6, 1700485571);
            d = md5_ii(d, a, b, c, x[i + 3], 10, -1894986606);
            c = md5_ii(c, d, a, b, x[i + 10], 15, -1051523);
            b = md5_ii(b, c, d, a, x[i + 1], 21, -2054922799);
            a = md5_ii(a, b, c, d, x[i + 8], 6, 1873313359);
            d = md5_ii(d, a, b, c, x[i + 15], 10, -30611744);
            c = md5_ii(c, d, a, b, x[i + 6], 15, -1560198380);
            b = md5_ii(b, c, d, a, x[i + 13], 21, 1309151649);
            a = md5_ii(a, b, c, d, x[i + 4], 6, -145523070);
            d = md5_ii(d, a, b, c, x[i + 11], 10, -1120210379);
            c = md5_ii(c, d, a, b, x[i + 2], 15, 718787259);
            b = md5_ii(b, c, d, a, x[i + 9], 21, -343485551);
            
            a = safe_add(a, olda);
            b = safe_add(b, oldb);
            c = safe_add(c, oldc);
            d = safe_add(d, oldd);
        }
        return [a, b, c, d];
    }
    
    function md5_cmn(q, a, b, x, s, t) {
        return safe_add(bit_rol(safe_add(safe_add(a, q), safe_add(x, t)), s), b);
    }
    
    function md5_ff(a, b, c, d, x, s, t) {
        return md5_cmn((b & c) | ((~b) & d), a, b, x, s, t);
    }
    
    function md5_gg(a, b, c, d, x, s, t) {
        return md5_cmn((b & d) | (c & (~d)), a, b, x, s, t);
    }
    
    function md5_hh(a, b, c, d, x, s, t) {
        return md5_cmn(b ^ c ^ d, a, b, x, s, t);
    }
    
    function md5_ii(a, b, c, d, x, s, t) {
        return md5_cmn(c ^ (b | (~d)), a, b, x, s, t);
    }
    
    function safe_add(x, y) {
        const lsw = (x & 0xFFFF) + (y & 0xFFFF);
        const msw = (x >> 16) + (y >> 16) + (lsw >> 16);
        return (msw << 16) | (lsw & 0xFFFF);
    }
    
    function bit_rol(num, cnt) {
        return (num << cnt) | (num >>> (32 - cnt));
    }
    
    str = utf8Encode(str);
    return binl2hex(binlMD5(str2binl(str), str.length * 8));
}

// HTTP请求函数
async function makeRequest(url, options = {}) {
    const token = await storageAPI.get('token') || '';
    const requestTime = Math.floor(Date.now() / 1000);
    
    // 生成请求签名 - 使用工作的签名构造方式
    const pathname = new URL(API_BASE_URL).pathname;
    const urlPath = url.split('?')[0];
    const signString = urlPath + token + requestTime + API_SECRET;
    const requestSign = md5(signString);
    
    const headers = {
        'Content-Type': 'application/json',
        'Request-Time': requestTime.toString(),
        'Authorization': token,
        'Request-Sign': requestSign,
        ...options.headers
    };
    
    try {
        const response = await fetch(API_BASE_URL + url, {
            method: options.method || 'GET',
            headers: headers,
            body: options.body ? JSON.stringify(options.body) : undefined
        });
        
        if (!response.ok) {
            throw new Error(`HTTP ${response.status}: ${response.statusText}`);
        }
        
        const data = await response.json();
        
        if (data.code === 200) {
            return data;
        } else {
            throw new Error(data.message || `API错误 (code: ${data.code})`);
        }
    } catch (error) {
        console.error('API request failed:', error);
        throw error;
    }
}

// 消息处理器
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
    handleMessage(message).then(sendResponse).catch(error => {
        sendResponse({ type: 'error', data: error.message });
    });
    return true; // 保持消息通道开放
});

async function handleMessage(message) {
    try {
        switch (message.type) {
            case 'updateData':
                const updateResult = await makeRequest(`/version_info?version=${message.param}`);
                return { type: message.type, data: updateResult.data };
                
            case 'indexData':
                const indexResult = await makeRequest('/index_data');
                return { type: message.type, data: indexResult.data };
                
            case 'qrcode':
                const qrResult = await makeRequest('/qr_code');
                return { type: message.type, data: qrResult.data };
                
            case 'scannerCodeState':
                const scanResult = await makeRequest(`/qr_code_state/${message.param}`);
                return { type: message.type, data: scanResult.data };
                
            case 'accountCollect':
                const accountResult = await makeRequest('/account_collect');
                return { type: message.type, data: accountResult.data };
                
            case 'incomeLineData':
                const lineResult = await makeRequest(`/income_line_data${message.param}&date_type=day`);
                return { type: message.type, data: lineResult.data };
                
            case 'accountList':
                let accountUrl = '/user_account';
                if (message.param) {
                    accountUrl += '?from=check_plug_target_account';
                }
                const listResult = await makeRequest(accountUrl);
                return { type: message.type, data: listResult.data };
                
            case 'fundData':
                const fundResult = await makeRequest(`/fund_hold?account_id=${message.param}`);
                return { type: message.type, data: fundResult.data, update: message.tip };
                
            case 'incomeData':
                let incomeUrl = `/income_data?account_id=${message.param}`;
                if (message.id === 'all') {
                    incomeUrl = '/income_data?collect=true';
                }
                const incomeResult = await makeRequest(incomeUrl);
                return { type: message.type, data: incomeResult.data };
                
            case 'fundSearch':
                let searchUrl = `/search_fund?keyword=${message.param}`;
                if (message.id) {
                    searchUrl += `&account_id=${message.id}`;
                }
                const searchResult = await makeRequest(searchUrl);
                return { type: message.type, data: searchResult.data };
                
            case 'addFund':
                const addResult = await makeRequest('/fund_hold', { method: 'POST', body: message.param });
                return { type: message.type, data: addResult.data };
                
            case 'removeFund':
                const removeResult = await makeRequest(`/remove_fund_hold${message.param}`, { method: 'DELETE' });
                return { type: message.type, data: removeResult.data };
                
            case 'noticeBar':
                const noticeResult = await makeRequest('/notice');
                return { type: message.type, data: noticeResult.data };
                
            case 'openInterval':
                startBadgeUpdate();
                return { type: 'success' };
                
            case 'getBadgeData':
                await updateBadgeData();
                startBadgeUpdate();
                return { type: 'success' };
                
            case 'hiddenBadgeData':
                hideBadge();
                stopBadgeUpdate();
                return { type: 'success' };
                
            case 'closeInterval':
                stopBadgeUpdate();
                return { type: 'success' };
                
            case 'updateAccountData':
                accountData = message.param;
                updateBadgeDisplay();
                return { type: 'success' };
                
            case 'updateIndexData':
                accountData.index_data = message.param;
                updateBadgeDisplay();
                return { type: 'success' };
                
            case 'updateEarnData':
                accountData.today_income = message.param.earn;
                accountData.today_income_rate = message.param.earn_rate;
                updateBadgeDisplay();
                return { type: 'success' };
                
            case 'chooseShowBadge':
                chooseBadgeDisplay(message.param);
                return { type: 'success' };
                
            default:
                return { type: 'unknown', message: 'Unknown message type' };
        }
    } catch (error) {
        console.error('Error handling message:', error);
        return { type: 'error', url_desc: `处理${message.type}请求`, data: error.message };
    }
}

// Badge相关功能
let updateInterval = null;
let accountData = {};

function hideBadge() {
    chrome.action.setBadgeText({ text: "" });
    chrome.action.setTitle({ title: "支持一键导入，最方便的基金实时收益查看工具。" });
}

function startBadgeUpdate() {
    if (!updateInterval) {
        updateInterval = setInterval(() => {
            if (isMarketOpen()) {
                updateBadgeData();
            }
        }, 30000);
    }
}

function stopBadgeUpdate() {
    if (updateInterval) {
        clearInterval(updateInterval);
        updateInterval = null;
    }
}

async function updateBadgeData() {
    try {
        const response = await makeRequest('/account_collect');
        accountData = response.data;
        updateBadgeDisplay();
    } catch (error) {
        console.error('Failed to update badge data:', error);
    }
}

function updateBadgeDisplay() {
    chooseBadgeDisplay(0); // 默认显示收益金额
}

function chooseBadgeDisplay(index) {
    const income = accountData.today_income || 0;
    const incomeRate = accountData.today_income_rate || 0;
    
    let color = "#aeb5c4";
    let text = "";
    let isNegative = false;
    
    if (index === 0) {
        // 显示收益金额
        color = income > 0 ? "#FC4E50" : income < 0 ? "#07B360" : "#aeb5c4";
        if (income < 0) isNegative = true;
        text = formatNumber(Math.abs(income).toString());
    } else if (index === 1) {
        // 显示收益率
        color = incomeRate > 0 ? "#FC4E50" : incomeRate < 0 ? "#07B360" : "#aeb5c4";
        text = parseFloat(incomeRate).toFixed(2);
        if (incomeRate < 0) isNegative = true;
    }
    
    if (isNegative) text = "-" + text;
    
    chrome.action.setBadgeText({ text: text });
    chrome.action.setBadgeBackgroundColor({ color: color });
    chrome.action.setTitle({ title: generateTooltip() });
}

function formatNumber(num) {
    num = Math.floor(Number(num));
    if (num >= 1000 && num < 10000) {
        return (num / 1000).toFixed(1) + "k";
    } else if (num >= 10000) {
        return (num / 10000).toFixed(1) + "w";
    }
    return num.toString();
}

function generateTooltip() {
    let tooltip = "";
    const indexData = accountData.index_data || {};
    
    // 添加指数信息
    const indices = {
        "1.000001": "上证指数",
        "1.000300": "沪深300", 
        "0.399001": "深证成指",
        "0.399006": "创业板指"
    };
    
    for (const [code, name] of Object.entries(indices)) {
        if (indexData[code]) {
            const dir = indexData[code].dir;
            const rate = (dir && dir !== "NaN") ? parseFloat(dir).toFixed(2) : "0.00";
            tooltip += `${name} ${rate}%  `;
        }
    }
    
    tooltip += "\n\n";
    
    // 添加收益信息
    if (accountData.is_single_account_model) {
        tooltip += `当日收益   ${parseFloat(accountData.today_income || 0).toFixed(2)}\n`;
        tooltip += `当日收益率   ${parseFloat(accountData.today_income_rate || 0).toFixed(2)}%\n`;
    } else {
        if (accountData.account_data) {
            accountData.account_data.forEach(account => {
                tooltip += `${account.title}收益   ${parseFloat(account.today_income).toFixed(2)}   ${parseFloat(account.today_income_rate).toFixed(2)}%\n`;
            });
        }
        tooltip += `\n总收益    ${parseFloat(accountData.today_income || 0).toFixed(2)}   ${parseFloat(accountData.today_income_rate || 0).toFixed(2)}%`;
    }
    
    return tooltip;
}

function isMarketOpen() {
    const now = new Date();
    const day = now.getDay();
    if (day === 0 || day === 6) return false;
    
    const hour = now.getHours();
    const minute = now.getMinutes();
    
    return (hour === 9 && minute >= 30) || 
           (hour >= 10 && hour <= 11) ||
           (hour === 11 && minute <= 30) ||
           (hour === 13 && minute >= 30) ||
           (hour >= 14 && hour <= 15);
}

// 初始化
chrome.storage.local.get(['token'], (result) => {
    if (result.token) {
        updateBadgeData();
        startBadgeUpdate();
    }
});

// 定期检查时间戳
setInterval(() => {
    chrome.storage.local.get(['timeStamp'], (result) => {
        if (result.timeStamp && Date.now() - result.timeStamp >= 20000) {
            storageAPI.remove('timeStamp');
            startBadgeUpdate();
        }
    });
}, 10000);
